import { message } from "antd"

// 存储表格信息
export const setTableData = () => {
  let data = []
  if (!!getTableData()) {
    // 获取日期
    let datax = new Date()
    let nian = datax.getFullYear()
    let yue =
      datax.getMonth() + 1 < 10
        ? '0' + (datax.getMonth() + 1)
        : datax.getMonth() + 1
    let ri = datax.getDate() + 1
    // 类别
    let leiBie = ['早餐', '电子', '工资', '理财', '衣物', '其他']
    for (let i = 0; i < 50; i++) {
      let dataS // 处理好的时间格式
      // 通过判断语句判断是否应该修改月份
      if (ri === '01') {
        ri = 30
        yue = yue - 1
        yue = yue < 10 ? '0' + yue : yue
        dataS = `${nian + '年' + yue + '月' + ri + '日'}`
      } else {
        ri = ri - 1
        ri = ri < 10 ? '0' + ri : ri
        dataS = `${nian + '年' + yue + '月' + ri + '日'}`
      }
      // 随机事件
      const sj = leiBie[Math.ceil(Math.random() * leiBie.length - 1)]
      // 判断是否是收入
      let income
      if (sj === '工资' || sj === '理财') {
        income = true
      } else if (sj === '其他') {
        income = Math.ceil(Math.random() * 2) === 1
      } else {
        income = false
      }
      const sjMoney = 100 + i
      data.push({
        key: i + ``,
        date: dataS,
        money: !income ? -(100 + i) : 100 + i,
        sort: sj,
        notes: sj,
      })
    }
  } else {
    data.push(getTableData())
  }
  // 本地存储
  localStorage.setItem('BillData', JSON.stringify(data))
  return data
}

// 获取表格信息
export const getTableData = () => {
  const data = localStorage.getItem('BillData')
  return data
}

// 添加表格信息
export const addTableData = (data) => {
  let datax = []
  datax.push(JSON.parse(getTableData()))
  datax[0].unshift(data)
  // 本地存储
  localStorage.setItem('BillData', JSON.stringify(datax[0]))
  return data
}

// 删除表格信息
export const delTableData = (data) => {
  // 本地存储
  localStorage.setItem('BillData', JSON.stringify(data))

}


// 修改表格信息
export const changeTableData = (data) => {
  const dataAll = JSON.parse(getTableData())
  const dataX = dataAll.map(item => {
    if (item.key === data.key) {
      return data
    } else {
      return item
    }
  })
  // 本地存储
  localStorage.setItem('BillData', JSON.stringify(dataX))
  return data
}
